package com.zzy.util.file;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.util.HashMap;
import java.util.Map;

import com.zzy.util.DataMapUtil;
import freemarker.template.*;

public class UtilDocFrameMaker {
    private Configuration configure = null;
    public UtilDocFrameMaker(){
        configure= new Configuration();
        configure.setDefaultEncoding("utf-8");
    }
    /**
     * 根据Doc模板生成word文件
     * @param dataMap Map 需要填入模板的数据
     * @param downloadType 文件名称
     * @param savePath 保存路径
     */
    public void createDoc(Map<String, Object> dataMap, String downloadType, String savePath){
        try{
            //加载需要装填的模板
            Template template  = null;
            //加载模板文件
            configure.setClassForTemplateLoading(this.getClass(),"/com/favccxx/secret/templates");
            //设置对象包装器
            configure.setObjectWrapper(new DefaultObjectWrapper());
            //设置异常处理器
            configure.setTemplateExceptionHandler(TemplateExceptionHandler.IGNORE_HANDLER);
            //定义Template对象,注意模板类型名字与downloadType要一致
            template= configure.getTemplate(downloadType + ".xml");
            //输出文档
            File outFile = new File(savePath);
            Writer out = null;
            out= new BufferedWriter(new OutputStreamWriter(new FileOutputStream(outFile),"utf-8"));
            template.process(dataMap,out);
            outFile.delete();
        }catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 根据下载类型获取需要传递的Map参数
     * @param oid 对象Id
     * @param downloadType 下载类型
     */
    private Map<String, Object> getDataMap(String oid,String downloadType){
        Map<String, Object> dataMap = new HashMap<String, Object>();
        if("Parameter1".equals(downloadType)){
            dataMap = DataMapUtil.setObjToMap("");
        }else{
            dataMap = DataMapUtil.setObjToMap("");
        }
        return dataMap;
    }

}